// Tencent is pleased to support the open source community by making ncnn available.
//
// Copyright (C) 2021 THL A29 Limited, a Tencent company. All rights reserved.
//
// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
// in compliance with the License. You may obtain a copy of the License at
//
// https://opensource.org/licenses/BSD-3-Clause
//
// Unless required by applicable law or agreed to in writing, software distributed
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
// specific language governing permissions and limitations under the License.

#ifndef NDKCAMERA_H
#define NDKCAMERA_H

#include <android/looper.h>
#include <android/native_window.h>
#include <android/sensor.h>
#include <camera/NdkCameraDevice.h>
#include <camera/NdkCameraManager.h>
#include <camera/NdkCameraMetadata.h>
#include <media/NdkImageReader.h>
#include <opencv2/core/core.hpp>
#include "JvmCallback.h"
#include "yolo.h"
#include "safe_queue.h"
#include "ObjPacket.h"

class NdkCamera {
public:
    NdkCamera();

    virtual ~NdkCamera();

    // facing 0=front 1=back
    int open(int camera_facing = 0,int width = 480,int height = 640);

    void close();

    virtual void on_image_record(unsigned char *nv21, int width, int height) const;

    virtual void on_image(const unsigned char *nv21, int nv21_width, int nv21_height) const;

    virtual void
    on_image_callback(const unsigned char *nv21, int nv21_width, int nv21_height);

public:
    int camera_facing;
    int camera_orientation;
    bool isRecording = false;

private:
    ACameraManager *camera_manager;
    ACameraDevice *camera_device;
    AImageReader *image_reader;
    ANativeWindow *image_reader_surface;
    ACameraOutputTarget *image_reader_target;
    ACaptureRequest *capture_request;
    ACaptureSessionOutputContainer *capture_session_output_container;
    ACaptureSessionOutput *capture_session_output;
    ACameraCaptureSession *capture_session;

};

class NdkCameraWindow : public NdkCamera {
private:
    ASensorManager *sensor_manager;
    mutable ASensorEventQueue *sensor_event_queue;
    const ASensor *accelerometer_sensor;
    ANativeWindow *win;
    SafeQueue<ObjPacket *> obj_queue;
    pthread_t try_data_t;
    long first_time = 0;
    long last_time = 0;
public:
    NdkCameraWindow();

    virtual ~NdkCameraWindow();

    void set_window(ANativeWindow *win);

    virtual void on_image_render(cv::Mat &rgb, std::vector<Object> &objects) const;

    virtual void on_image_record(unsigned char *nv21, int width, int height) const;

    virtual void on_image_callback(const unsigned char *nv21, int nv21_width, int nv21_height);

    virtual void on_image(const unsigned char *nv21, int nv21_width, int nv21_height) const;

    void on_save_callback(cv::Mat &rgb, std::vector<Object> &objects);

    virtual void startTryData();
    virtual void stopTryData();

    void save_yolo_data();


    static void releaseObjPacket(ObjPacket **packet) {
        if (packet != nullptr && *packet != nullptr) {
            delete *packet;
            *packet = nullptr;
        }
    }
    static void dropPacket(queue<ObjPacket *> &q){
        if (!q.empty()){
            ObjPacket * p = q.front();
            NdkCameraWindow::releaseObjPacket(&p);
            q.pop();
        }
    }

public:
    mutable int accelerometer_orientation;
    JvmCallback *jvm_callbacks;
    std::string video_path;
    std::string dir_path;
    long id = 0;


};

#endif // NDKCAMERA_H
